#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：casing
@File ：proc_excel.py
@IDE ：PyCharm
@Author ：GaoYijia
@Date ：2024/6/4 11:41
"""

import pandas as pd
import os


def find_common_books(directory, output_file):
    # 获取指定目录中的所有Excel文件
    files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith('.xlsx')]

    if len(files) < 2:
        print("目录中没有足够的Excel文件进行比较。")
        return

    # 读取第一个文件
    common_books = pd.read_excel(files[0])

    # 逐个比较其余的文件
    for file in files[1:]:
        df = pd.read_excel(file)
        common_books = pd.merge(common_books, df, on='书名')

    # 将结果写入新的Excel文件
    common_books.to_excel(output_file, index=False)


if __name__ == "__main__":
    # 指定数据目录和输出文件路径
    directory = 'data' # 原始的excel文件放在 data 文件夹下
    output_file = 'common_books.xlsx'   #合并后的文件

    find_common_books(directory, output_file)
    print(f"Common books saved to {output_file}")


